# pip install zhipuai 请先在终端进行安装

import zhipuai

zhipuai.api_key = "41805a5412196acfdd9d63a905f9039a.Z0TWy2173NoP1ZkI"
response = zhipuai.model_api.sse_invoke(
    model="chatglm_lite",
    prompt= [{"role":"user","content":"假设你是一个编剧，想象力丰富，请你写一个关于恐怖游轮的剧本，其中有5个角色，每个人都参与了游轮上发生的案件，故事包含有悬疑、恐怖元素。\n人物设定从“年龄、 性格， 长相、 成长背景、 经历、 特长、 相互纠葛”7个方面进行设定\n情节设定从以下4个方面设定：\n1、主线：核心事件及发生经过简介\n2、辅线：辅助事件及发生经过简介\n3、暗线：隐藏在故事中的线索\n4、结构概括：从开局、 转折、高潮、结局4个方面设定"}],
    temperature= 0.9,
    top_p= 0.7,
    incremental=True
)

for event in response.events():
    if event.event == "add":
        print(event.data, end="")
    elif event.event == "error" or event.event == "interrupted":
        print(event.data, end="")
    elif event.event == "finish":
        print(event.data)
        print(event.meta, end="")
    else:
        print(event.data, end="")